import pygame
from pygame.sprite import Sprite


class Spacecraft(Sprite):

    def __init__(self, settings, screen):
        """初始化飞船并设置其初始位置"""
        super().__init__()
        self.screen = screen
        self.settings = settings

        # 加载飞船图像并获取其外接矩形
        self.image = pygame.image.load('images_source/spacecraft1.png')
        self.rect = self.image.get_rect()
        self.screen_rect = screen.get_rect()

        # 将每艘新飞船放在屏幕底部正中央
        self.rect.centerx = self.screen_rect.centerx
        self.rect.bottom = self.screen_rect.bottom

        # 在飞船center属性中存储浮点值
        self.center = float(self.rect.centerx)
        self.bottom = float(self.rect.centery)

        # 移动标志
        self.moving_right = False
        self.moving_left = False
        self.moving_front = False
        self.moving_back = False

    def adjust_spacecraft_position(self):
        """根据移动标志调整飞船位置"""
        if self.moving_right and self.rect.right < self.screen_rect.right:
            self.center += self.settings.spacecraft_speed_factor
        if self.moving_left and self.rect.left > 0:
            self.center -= self.settings.spacecraft_speed_factor
        if self.moving_front and self.rect.top > 0:
            self.bottom -= self.settings.spacecraft_speed_factor
        if self.moving_back and self.rect.bottom < self.screen_rect.bottom:
            self.bottom += self.settings.spacecraft_speed_factor

        # 更新rect对象
        self.rect.centerx = self.center
        self.rect.centery = self.bottom

    def draw_spacecraft(self):
        """在指定位置绘制飞船"""
        self.screen.blit(self.image, self.rect)

    def reset(self, settings, screen):
        """让飞船回到初始位置"""
        self.__init__(settings, screen)

